package com.gzhryc.common;

import org.apache.commons.io.FilenameUtils;

public class FileTools {

    public static boolean isImageFile(String fileName){
        String extensionName = FilenameUtils.getExtension(fileName);
        if(StringTools.isNotBlank(extensionName)){
            return isImageExtensionName(extensionName);
        }
        return false;
    }

    public static boolean isImageExtensionName(String extensionName) {
        extensionName = extensionName.toLowerCase();
        if ("png".equals(extensionName) || "jpg".equals(extensionName) || "jpeg".equals(extensionName)
                || "gif".equals(extensionName) || "bmp".equals(extensionName)) {
            return true;
        }
        return false;
    }

    public static boolean isVideoFile(String fileName){
        String extensionName = FilenameUtils.getExtension(fileName);
        if(StringTools.isNotBlank(extensionName)){
            extensionName = extensionName.toLowerCase();
            if("mp4".equals(extensionName) || "avi".equals(extensionName) || "wmv".equals(extensionName)){
                return true;
            }
        }
        return false;
    }
    
    public static boolean isCompressFile(String fileName){
        String extensionName = FilenameUtils.getExtension(fileName);
        if(StringTools.isNotBlank(extensionName)){
            extensionName = extensionName.toLowerCase();
            if("zip".equals(extensionName) || "tar".equals(extensionName)){
                return true;
            }
        }
        return false;
    }
}
